import { Badge as AntBadge } from 'antd'

export type BadgeProps = React.ComponentProps<typeof AntBadge>

// interface BadgeProps {
//   /**
//    * 自定义小圆点的颜色
//    */
//   color?: string
//   /**
//    * 展示的数字，大于 overflowCount 时显示为 ${overflowCount}+，为 0 时隐藏
//    */
//   count?: React.ReactNode
//   /**
//    * 不展示数字，只有一个小红点
//    * @default false
//    */
//   dot?: boolean
//   /**
//    * 设置状态点的位置偏移
//    */
//   offset?: [number | string, number | string]
//   /**
//    * 展示封顶的数字值
//    * @default 99
//    */
//   overflowCount?: number
//   /**
//    * 当数值为 0 时，是否展示 Badge
//    * @default false
//    */
//   showZero?: boolean
//   /**
//    * 在设置了 count 的前提下有效，设置小圆点的大小
//    */
//   size?: 'default' | 'small'
//   /**
//    * 设置 Badge 为状态点
//    */
//   status?: 'success' | 'processing' | 'default' | 'error' | 'warning'
//   /**
//    * 在设置了 status 的前提下有效，设置状态点的文本
//    */
//   text?: React.ReactNode
//   /**
//    * 设置鼠标放在状态点上时显示的文字
//    */
//   title?: string
//   children?: React.ReactNode
// }

export const Badge = ({ children, ...restProps }: BadgeProps) => {
  return <AntBadge {...restProps}>{children}</AntBadge>
}
